from pydantic import BaseModel
from typing import Optional
# 定义 用户信息抽取Pydantic 模型
class UserInfo(BaseModel):
    birthday: str
    constellation: str # 星座
    user_question: str

# 定义消息状态模型
class ExtractionState(BaseModel):
    query: str
    answer: Optional[UserInfo] = None